Rehabilitation device based on electrical stimulation